package com.ph.generator;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

public class MyBatisPlusGenerator {

    private static final String author = "swallow";
    // account
//    private static final String outputDir = "F:\\learn\\Seate\\simple\\simple-account\\src\\main\\java";
//    private static final String[] tables = new String[] {"t_account"};
//    private static final String parentPackage = "com.ph.seate.account";
//    private static final String driverName = "com.mysql.jdbc.Driver";
//    private static final String dbUrl = "jdbc:mysql://ph.base.cn:3306/simple_account?useSSL=false";
    // order
//    private static final String outputDir = "F:\\learn\\Seate\\simple\\simple-order\\src\\main\\java";
//    private static final String[] tables = new String[] {"t_order"};
//    private static final String parentPackage = "com.ph.seata.order";
//    private static final String driverName = "com.mysql.jdbc.Driver";
//    private static final String dbUrl = "jdbc:mysql://ph.base.cn:3306/simple_order?useSSL=false";
    // storage
    private static final String outputDir = "F:\\learn\\Seate\\simple\\simple-storage\\src\\main\\java";
    private static final String[] tables = new String[] {"t_storage"};
    private static final String parentPackage = "com.ph.seata.storage";
    private static final String driverName = "com.mysql.jdbc.Driver";
    private static final String dbUrl = "jdbc:mysql://ph.base.cn:3306/simple_storage?useSSL=false";

    private static final String dbUsername = "root";
    private static final String dbPassword = "swallow";


    public static void main(String[] args) {
        //1. 全局配置
        GlobalConfig config = new GlobalConfig();
        // 是否支持AR模式
        config.setActiveRecord(true)
                // 作者
                .setAuthor(author)
                // 生成路径，最好使用绝对路径，window路径是不一样的
                //TODO  TODO  TODO  TODO
                .setOutputDir(outputDir)
                // 文件覆盖
                .setFileOverride(true)
                // 主键策略
                .setIdType(IdType.AUTO)

                .setDateType(DateType.ONLY_DATE)
                // 设置生成的service接口的名字的首字母是否为I，默认Service是以I开头的
                .setServiceName("%sService")

                //实体类结尾名称
                .setEntityName("%sDO")

                //生成基本的resultMap
                .setBaseResultMap(true)

                //不使用AR模式
                .setActiveRecord(false)

                //生成基本的SQL片段
                .setBaseColumnList(true);

        //2. 数据源配置
        DataSourceConfig dsConfig = new DataSourceConfig();
        // 设置数据库类型
        dsConfig.setDbType(DbType.MYSQL)
                .setDriverName(driverName)
                //TODO  TODO  TODO  TODO
                .setUrl(dbUrl)
                .setUsername(dbUsername)
                .setPassword(dbPassword);

        //3. 策略配置globalConfiguration中
        StrategyConfig stConfig = new StrategyConfig();

        //全局大写命名
        stConfig.setCapitalMode(true)
                // 去除前缀
                .setTablePrefix("t_")
                // 数据库表映射到实体的命名策略
                .setNaming(NamingStrategy.underline_to_camel)

                //使用lombok
                .setEntityLombokModel(true)

                //使用restcontroller注解
                .setRestControllerStyle(true)

                // 生成的表, 支持多表一起生成，以数组形式填写
                //TODO  TODO  TODO  TODO
                .setInclude(tables);

        //4. 包名策略配置
        PackageConfig pkConfig = new PackageConfig();
        pkConfig.setParent(parentPackage)
                .setMapper("mapper")
                .setService("service")
                .setController("controller")
                .setEntity("model")
                .setXml("mapper");

        //5. 整合配置
        AutoGenerator ag = new AutoGenerator();
        ag.setGlobalConfig(config)
                .setDataSource(dsConfig)
                .setStrategy(stConfig)
                .setPackageInfo(pkConfig);

        //6. 执行操作
        ag.execute();
        System.out.println("======= Done 相关代码生成完毕  ========");
    }
}